package com.example.d4_collection_set;

import java.util.Comparator;
import java.util.TreeSet;

/**
 * TreeSet源码分析
 */
@SuppressWarnings({"all"})
public class TreeSetSource {
    public static void main(String[] args) {

        // 1.当我们使用无参构造器,创建TreeSet时,仍然是无序的
        // 2.希望添加的元素是,按照字符串大小来排序
        // 3.使用TreeSet 提供的一个构造器,可以传入一个比较器(匿名内部类)并指定排序规则
        // 4.看源码

        //源码分析：
        /*
        1. 构造器把传入的比较器(comparator)对象,赋给了 TreeSet的底层的 TreeMap的属性this.comparator

         public TreeMap(Comparator<? super K> comparator) {
                this.comparator = comparator;
            }
         2. 在 调用 treeSet.add("tom"), 在底层会执行到
              Comparator<? super K> cpr = comparator;//把我们传入的匿名对象赋给cpr
              因为我们传入了一个匿名内部类 所以comparator不为空 就会进入 do while循环
             if (cpr != null) {//cpr 就是我们的匿名内部类(对象)
                do {
                    parent = t;
                    cmp = cpr.compare(key, t.key);//动态绑定到我们的匿名内部类(对象)compare
                    if (cmp < 0)
                        t = t.left;
                    else if (cmp > 0)
                        t = t.right;
                    else //如果相等,即返回0,这个Key就没有加入
                        return t.setValue(value);
                } while (t != null);
            }
         */

        // TreeSet treeSet = new TreeSet();
        // 使用一个有参数的构造器,传入一个比较器,就可以按照我们指定的规则来进行排序
        TreeSet treeSet = new TreeSet(new Comparator() {
            @Override
            public int compare(Object o1, Object o2) {
                //下面 调用String的 compareTo方法进行字符串大小比较
                //要求加入的元素,按照长度大小排序
                //return ((String) o2).compareTo((String) o1);
                //底层会有do while循化依次进行比较  根据我们全入的方法而定
                return ((String) o1).length() - ((String) o2).length();
            }
        });
        // 添加数据.
        treeSet.add("jack");
        treeSet.add("tom"); // 3
        treeSet.add("sp");
        treeSet.add("a");
        treeSet.add("abc"); // 3因为我们是按照字符串的长度进行比较,因此abc不会添加成功
        System.out.println("treeSet=" + treeSet);
    }
}